import tensorflow as tf
from keras.layers import Dense
from keras import Sequential
import os

if __name__ == '__main__':
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

    # 通过 Sequential 自动将层连接到一起
    model = Sequential(
        [
            Dense(units=256, activation='relu'),
            Dense(units=128, activation='relu'),
            Dense(units=64, activation='relu'),
            Dense(units=10, activation='relu')
        ]
    )

    n = 6
    input = tf.random.normal(shape=[n, 28 * 28])
    output = model(input)

    print(output)



